from pydantic import BaseModel, ConfigDict, Field
from pydantic.alias_generators import to_camel
from pydantic_validation_decorator import NotBlank
from typing import Optional
from module_admin.annotation.pydantic_annotation import as_query




class StudentModel(BaseModel):
    """
    学生表对应pydantic模型
    """
    model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)

    name: Optional[str] = Field(default=None, description='姓名')
    id: Optional[int] = Field(default=None, description='ID')

    @NotBlank(field_name='name', message='姓名不能为空')
    def get_name(self):
        return self.name


    def validate_fields(self):
        self.get_name()




class StudentQueryModel(StudentModel):
    """
    学生不分页查询模型
    """

    begin_time: Optional[str] = Field(default=None, description='开始时间')
    end_time: Optional[str] = Field(default=None, description='结束时间')


@as_query
class StudentPageQueryModel(StudentQueryModel):
    """
    学生分页查询模型
    """

    page_num: int = Field(default=1, description='当前页码')
    page_size: int = Field(default=10, description='每页记录数')


class DeleteStudentModel(BaseModel):
    """
    删除学生模型
    """

    model_config = ConfigDict(alias_generator=to_camel)

    ids: str = Field(description='需要删除的ID')
